package com.runa.protocol.dto;

import com.runa.monitor.comm.dto.Node;
import com.runa.monitor.comm.dto.Valve;
import com.runa.monitor.comm.entity.CommNetEquipment;
import com.runa.monitor.comm.entity.CommValve;
import lombok.Data;

import java.util.ArrayList;
import java.util.List;

/**
 * 2024/9/11 11:47
 * zhangYB
 */
@Data
public class BatchValveDTO {

    private List<CommNetEquipment> commNetEquipmentList = new ArrayList<>();

    private List<Node> fourGValves = new ArrayList<>();

    public void setCommNetEquipmentList(List<CommNetEquipment> commNetEquipmentList) {
        this.commNetEquipmentList = commNetEquipmentList;
    }

    public void addFourGValve(CommValve commValve) {
        fourGValves.add(Valve.getValve(commValve));
    }

    public void setFourGValves(List<Node> fourGValves) {
        this.fourGValves = fourGValves;
    }


}
